#pragma once
#include <functional>
#include <iostream>
#include <cstdint>
#include <cstring>
#include <fmt/core.h>
#include <pthread.h>
#include <string>
#include <unistd.h>

namespace ThreadModlue
{
    static uint32_t number = 1;

    template <typename T>
    class Thread
    {
        using func_t = std::function<void(T)>;

    private:
        /**
         * @brief 启用线程分离功能
         *
         * 该函数用于将当前线程设置为分离状态。在分离状态下，线程结束时资源会被自动释放。
         * 当线程被分离后，将输出提示信息 "线程被分离了"。
         *
         * @note 调用此函数后，_isdetach 成员变量将被设置为 true。
         */
        void EnableDetach()
        {
            println("线程被分离了"); // 输出提示信息，表示线程已被分离
            _isdetach = true;        // 将 _isdetach 成员变量设置为 true，表示线程处于分离状态
        }
        /**
         * @brief 启用运行状态
         *
         * 该函数用于将对象的运行状态设置为启用。调用此函数后，
         * 对象的_isrunning成员变量将被设置为true，表示对象处于运行状态。
         */
        void EnableRunning()
        {
            _isrunning = true;
        }
        /**
         * @brief 线程执行例程函数
         *
         * 该函数是线程启动后执行的例程函数，用于处理线程的主要任务。
         *
         * @param args 传递给线程例程的参数，通常是一个指向线程对象的指针。
         * @return 返回值为 nullptr，表示线程执行结束。
         */
        static void *Routine(void *args)
        {
            // 将传入的参数转换为 Thread<T> 类型的指针
            Thread<T> *self = static_cast<Thread<T> *>(args);

            // 调用线程对象的 EnableRunning 方法，设置线程为运行状态
            self->EnableRunning();

            // 检查线程是否设置为分离状态，如果是，则调用 Detach 方法进行分离
            if (self->_isdetach)
                self->Detach();

            // 调用线程对象存储的函数指针，并传递数据参数
            self->_func(self->_data);

            // 返回 nullptr，表示线程执行结束
            return nullptr;
        }

    public:
      Thread(const Thread &) = delete;
      Thread(Thread &&) = delete;
      Thread &operator=(const Thread &) = delete;
      Thread &operator=(Thread &&) = delete;
      /**
       * @brief 构造函数，用于初始化线程对象。
       *
       * @param func 线程执行的函数指针。
       * @param data 传递给线程函数的参数。
       */
      Thread(func_t func, T data)
          : _tid(0),           // 初始化线程ID为0。
            _isdetach(false),  // 初始化线程是否已分离为false。
            _isrunning(false), // 初始化线程是否正在运行为false。
            _func(func),       // 初始化线程执行的函数指针。
            _data(data)        // 初始化传递给线程函数的参数。
      {
        _name = "thread-" +
                std::to_string(number++); // 生成线程名称，并递增线程编号。
        }
        /**
         * @brief 分离当前线程
         *
         * 该函数用于分离当前线程，使其在完成后自动释放资源。
         * 如果线程已经处于分离状态，则函数直接返回。
         * 如果线程正在运行，则调用pthread_detach函数进行分离。
         * 最后，调用EnableDetach函数启用分离功能。
         */
        void Detach()
        {
            // 检查线程是否已经处于分离状态
            if (_isdetach)
            {
                return;
            }
            // 检查线程是否正在运行
            if (_isrunning)
                // 调用pthread_detach函数分离线程
                pthread_detach(_tid);
            // 启用分离功能
            EnableDetach();
        }
        /**
         * @brief 启动线程。
         *
         * 该函数用于启动一个新的线程来执行指定的任务。如果线程已经处于运行状态，
         * 则返回false。否则，尝试创建一个新的线程，并返回创建结果。
         *
         * @return 如果线程创建成功，返回true；否则返回false。
         */
        bool Start()
        {
            // 检查线程是否已经在运行
            if (_isrunning)
                return false;

            // 创建新线程，执行Routine函数，并将this作为参数传递
            int n = pthread_create(&_tid, nullptr, Routine, this);

            // 检查线程创建是否成功
            if (n != 0)
            {
                // 如果创建失败，输出错误信息
                std::cerr << "creat thread error: " << strerror(n) << std::endl;
                return false;
            }
            else
            {
                // 如果创建成功，输出成功信息
                println("{} start success", _name);
                return true;
            }
        }
        /**
         * @brief 停止正在运行的线程。
         *
         * 该函数用于停止一个正在运行的线程。首先检查线程是否正在运行，
         * 如果是，则尝试取消线程。如果取消线程成功，则将线程状态设置为停止，
         * 并输出停止信息；如果取消线程失败，则输出错误信息并返回失败状态。
         *
         * @return 如果线程成功停止，返回 true；否则返回 false。
         */
        bool Stop()
        {
            if (_isrunning)
            {
                // 尝试取消线程
                int n = pthread_cancel(_tid);
                if (n != 0)
                {
                    // 如果取消线程失败，输出错误信息
                    std::cerr << "create thread error: " << strerror(n) << std::endl;
                    return false;
                }
                else
                {
                    // 如果取消线程成功，更新线程状态并输出停止信息
                    _isrunning = false;
                    println("{} stop", _name);
                    return true;
                }
            }
            // 如果线程未运行，直接返回失败状态
            return false;
        }
        /**
         * @brief 尝试将当前线程与指定的线程进行连接（join）。
         *
         * 该函数首先检查目标线程是否已经被分离（detached）。如果已经被分离，则输出错误信息并返回。
         * 否则，调用pthread_join函数尝试将当前线程与目标线程进行连接。如果连接失败，输出错误信息；
         * 如果连接成功，输出成功信息。
         */
        void Join()
        {
            // 检查线程是否已经被分离
            if (_isdetach)
            {
                // 输出错误信息：线程已经被分离，不能进行join
                println("你的线程已经是分离的了,不能进行join");
                return;
            }
            // 调用pthread_join函数尝试进行线程连接
            int n = pthread_join(_tid, &res);
            // 检查pthread_join的返回值
            if (n != 0)
                // 如果返回值不为0，表示连接失败，输出错误信息
                std::cerr << "create thread error: " << strerror(n) << std::endl;
            else
                // 如果返回值为0，表示连接成功，输出成功信息
                println("join success");
        }
        ~Thread() = default;

    private:
        pthread_t _tid;    //  线程ID
        std::string _name; //  线程名称
        bool _isdetach;    //  是否分离
        bool _isrunning;   //  是否正在运行
        void *res;         //  返回值
        func_t _func;      //  线程函数
        T _data;           //  传递给线程函数的数据
    };
} // namespace ThreadModlue
